package com.github.yangyishe.p500;

import com.github.yangyishe.ListNode;
import com.github.yangyishe.TreeNode;

import java.util.HashMap;
import java.util.Map;

/**
 * 437. 路径总和 III
 * https://leetcode.cn/problems/path-sum-iii/description/?envType=study-plan-v2&envId=top-100-liked
 * todo 已解决，但效率较低，以后再优化
 *
 * 给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
 *
 * 路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。
 *
 *
 *
 * 示例 1：
 *
 *
 *
 * 输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
 * 输出：3
 * 解释：和等于 8 的路径有 3 条，如图所示。
 * 示例 2：
 *
 * 输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
 * 输出：3
 *
 *
 * 提示:
 *
 * 二叉树的节点个数的范围是 [0,1000]
 * -109 <= Node.val <= 109
 * -1000 <= targetSum <= 1000
 */
public class Problem437 {
    public static void main(String[] args) {
//        ListNode listNode = ListNode.getInstance(new int[]{1, 2, 4, 5, -1, -5, 3, 3, 6, 4});
//
//        Problem437 problem437 = new Problem437();
//        int pathSum = problem437.pathSum(listNode, 6);
//        System.out.println("pathSum = " + pathSum);


//        TreeNode treeNode = TreeNode.getInstanceNew(new Integer[]{10, 5, -3, 3, 2, null, 11, 3, -2, null, 1});
//        TreeNode treeNode = TreeNode.getInstanceNew(new Integer[]{1000000000,1000000000,null,294967296,null,1000000000,null,1000000000,null,1000000000});
        TreeNode treeNode = TreeNode.getInstanceNew(new Integer[]{0,1,1});
        int target=1;

        Problem437 problem437 = new Problem437();
        int pathSum = problem437.pathSum(treeNode, target);
        System.out.println("pathSum = " + pathSum);


    }

    private int sumCount=0;

    /**
     * 思路：
     * 先从简单的开始考虑，假如是队列的情况下，如何确定count?
     * 从开始遍历的话，有两种情况：当前节点参与计算；当前节点不参与计算。
     * 根据两种情况的不同，分裂为不同的分支，以待求和为记录（实际可以整理为一个map）
     *
     * @param root
     * @param targetSum
     * @return
     */
    public int pathSum(TreeNode root, int targetSum) {
        Map<Long,Integer> target2CountMap=new HashMap<>();
        target2CountMap.put((long) targetSum,1);
        calcPathSum(root,targetSum,target2CountMap);

        return sumCount;
    }

    public void calcPathSum(TreeNode node,long targetSum,Map<Long,Integer> beforeMap){
        if(node==null){
            return;
        }
        if (beforeMap.get((long)node.val)!=null) {
            sumCount+=beforeMap.get((long)node.val);
//            beforeMap.remove(node.val);
        }
        Map<Long,Integer> target2CountMap=new HashMap<>();
        for (Long before : beforeMap.keySet()) {
            target2CountMap.put(before-node.val,beforeMap.get(before));
        }
        target2CountMap.put((long) targetSum,target2CountMap.getOrDefault(targetSum,0)+1);

        calcPathSum(node.left,targetSum,target2CountMap);
        calcPathSum(node.right,targetSum,target2CountMap);
    }


    /**
     * 链表模式版
     * 此处可以使用map
     * @param root
     * @param targetSum
     * @return
     */
    public int pathSum(ListNode root, int targetSum){
        Map<Integer,Integer> target2CountMap=new HashMap<>();
        target2CountMap.put(targetSum,1);
        calcPathSum(root,targetSum,target2CountMap);

        return sumCount;
    }



    public void calcPathSum(ListNode node, int targetSum, Map<Integer,Integer> beforeMap){
        if(node==null){
            return;
        }
        if (beforeMap.get(node.val)!=null) {
            sumCount+=beforeMap.get(node.val);
            beforeMap.remove(node.val);
        }
        Map<Integer,Integer> target2CountMap=new HashMap<>();
        for (Integer before : beforeMap.keySet()) {
            target2CountMap.put(before-node.val,beforeMap.get(before));
        }
        target2CountMap.put(targetSum,target2CountMap.getOrDefault(targetSum,0)+1);

        calcPathSum(node.next,targetSum,target2CountMap);
    }

}
